package com.liu.qqclient.Service;

import com.liu.qqcommon.Message;
import com.liu.qqcommon.MessageType;

import java.io.*;

/**
 * 该类完成文件传输
 */
public class FileClientService {
    /**
     *
     * @param src  源文件
     * @param dest  传输到对方哪个目录下
     * @param senderId  发送者
     * @param getterId  接收者
     */
    public void sendFileToOne(String src,String dest,String senderId,String getterId){
        //读取src的文件，文件必须存在。
        //封装到Message对象
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_FILE_MES );
        message.setSender(senderId);
        message.setGetter(getterId);
        message.setSrc(src);
        message.setData(dest);

        //需要将文件读取
        FileInputStream fileInputStream = null;
        byte[] fileBytes = new byte[(int)new File(src).length()];

        try {
            fileInputStream = new FileInputStream(src);
            fileInputStream.read(fileBytes);//将src文件读入到程序的字节数组中
            //将文件对应的字节设置到Message对象中
            message.setFileBytes(fileBytes);
        } catch (Exception e) {
            e.printStackTrace();
        }finally{
            //关闭流
            if(fileInputStream != null){
                try {
                    fileInputStream.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
        System.out.println("\n" + senderId + "给" + getterId + "发送文件" + src + "到" + dest);


        try {
            ObjectOutputStream oos = new ObjectOutputStream(ManageClientConnectServerThread.getClientConnectServerThread(senderId).getSocket().getOutputStream());
            oos.writeObject(message);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}
